import pickle
import pandas as pd
import quandl

api_key = "rFsSehe51RLzREtYhLfo"


def state_list():
    fifty_states = pd.read_html("https://simple.wikipedia.org/wiki/List_of_U.S._states")
    return fifty_states[0][0][1:]


def initial_state_data():
    states = state_list()
    main_df = pd.DataFrame()

    for abbv in states:
        query = "FMAC/HPI_" + str(abbv)
        df = quandl.get(query, authtoken=api_key)
        df.columns = [str(abbv)]
        if main_df.empty:
            main_df = df
        else:
            main_df = main_df.join(df)

    print(main_df.head())

    pickle_out = open("fifty_states.pickle", "wb")
    pickle.dump(main_df, pickle_out)
    pickle_out.close()


# initial_state_data()

pickle_in = open("fifty_states.pickle", "rb")
HPI_data = pickle.load(pickle_in)

print(HPI_data)
